import pandas as pd
from sqlalchemy import create_engine

# 假设你已经有了一个DataFrame，名为df


moneyflow_path = r'F:\Personal\data\merged\ths_member.csv'
df = pd.read_csv(moneyflow_path)

# df.rename(columns={'rank': 'ranking'}, inplace=True)
# df.drop(columns=['rank_reason'], inplace=True)

# 数据库连接信息
config = {
    "HOST": "192.168.200.179",
    "PORT": 43306,
    "USER": "qi0907",
    "PASSWORD": "Xuf0gQQzwPtw0D#6",
    "DATABASE": "XBBASE"  # 确保添加你的数据库名
}

# 创建数据库连接字符串
db_string = f"mysql+pymysql://{config['USER']}:{config['PASSWORD']}@{config['HOST']}:{config['PORT']}/{config['DATABASE']}"

# 创建数据库引擎
engine = create_engine(db_string)

# 确保你的DataFrame的列名与stk_zt表的列名相匹配
# 如果列名不匹配，你可能需要重命名列或修改表结构

# 写入DataFrame到MySQL数据库中的stk_zt表
df.to_sql('ths_member', con=engine, if_exists='append', index=False)

# 注意：if_exists='append'表示如果表已存在，则追加数据到表中
# index=False表示不将DataFrame的索引作为一列写入数据库